package com.byzk.p2p.admin.base.util;

import java.io.InputStream;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Date;
import java.util.Iterator;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFClientAnchor;
import org.apache.poi.hssf.usermodel.HSSFComment;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFPatriarch;
import org.apache.poi.hssf.usermodel.HSSFRichTextString;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.hssf.util.HSSFColor;
import org.w3c.dom.Document;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;

/**
 * 公用导出方法		
 * @author suker.zhou
 *
 * @param <T>
 */
public class ExcelUtil<T> {

	private String xmlPath = "conf/export/Export-";

	/**
	 * 获取表头
	 * @param t
	 * @return
	 */
	@SuppressWarnings("rawtypes")
	public String[] createHeaders(T t) {
		String a = t.getClass().toString();
		String b[] = a.split(" ");
		Class cla = null;
		try {
			cla = Class.forName(b[1]);
		} catch (ClassNotFoundException e) {
			e.printStackTrace();
		}
		// 获取对应xml路径
		String xmlName = xmlPath + cla.getSimpleName() + ".xml";
		List<String> list = new ArrayList<String>();
		try {
			InputStream in = ExcelUtil.class.getClassLoader().getResourceAsStream(xmlName);
			DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
			DocumentBuilder builder = factory.newDocumentBuilder();
			// 注意这里的Document是org.w3c.dom包下的
			Document doc = builder.parse(in); 
			NodeList nodeList=doc.getElementsByTagName("node").item(0).getChildNodes();
			for (int i = 0; i < nodeList.getLength(); i++) {
					if(nodeList.item(i).getNodeType()==Node.ELEMENT_NODE){
						String property=nodeList.item(i).getNodeName();
						list.add(doc.getElementsByTagName(property).item(0).getFirstChild().getNodeValue());
					}
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
		return list.toArray(new String[list.size()]);

	}

	/**
	 * 获取表头对应的字段
	 * @param t
	 * @return
	 */
	public String[] getHeaderFields(T t) {
		String a = t.getClass().toString();
		String b[] = a.split(" ");
		Class cla = null;
		try {
			cla = Class.forName(b[1]);
		} catch (ClassNotFoundException e) {
			e.printStackTrace();
		}
		// 获取对应xml路径
		String xmlName = xmlPath + cla.getSimpleName() + ".xml";
		List<String> list = new ArrayList<String>();
		try {
			InputStream in = ExcelUtil.class.getClassLoader().getResourceAsStream(xmlName);
			DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
			DocumentBuilder builder = factory.newDocumentBuilder();
			Document doc = builder.parse(in); // 注意这里的Document是org.w3c.dom包下的
			Node node = doc.getElementsByTagName("node").item(0);
			if(node.hasChildNodes()){
				NodeList nodeList=node.getChildNodes();
				for (int i = 0; i < nodeList.getLength(); i++) {
					if(nodeList.item(i).getNodeType()==Node.ELEMENT_NODE){
						String name = nodeList.item(i).getNodeName();
						// 获取字段
						list.add(name);
					}
				}	
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
		return list.toArray(new String[list.size()]);

	}
	/**
	 * 公用导出
	 * @param obj
	 * @param title
	 * @param dataset
	 * @param out
	 */
	public HSSFWorkbook exportExcel(T obj, Collection<T> dataset) {
		String[] headers = createHeaders(obj);
		// 声明一个工作薄

		HSSFWorkbook workbook = new HSSFWorkbook();

		// 生成一个表格

		HSSFSheet sheet = workbook.createSheet("sheet");

		// 设置表格默认列宽度为15个字节

		sheet.setDefaultColumnWidth(15);

		// 生成一个样式

		HSSFCellStyle style = workbook.createCellStyle();

		// 设置这些样式

		style.setFillForegroundColor(HSSFColor.SKY_BLUE.index);

		style.setFillPattern(HSSFCellStyle.SOLID_FOREGROUND);

		style.setBorderBottom(HSSFCellStyle.BORDER_THIN);

		style.setBorderLeft(HSSFCellStyle.BORDER_THIN);

		style.setBorderRight(HSSFCellStyle.BORDER_THIN);

		style.setBorderTop(HSSFCellStyle.BORDER_THIN);

		style.setAlignment(HSSFCellStyle.ALIGN_CENTER);

		// 生成一个字体

		HSSFFont font = workbook.createFont();

		font.setColor(HSSFColor.VIOLET.index);

		font.setFontHeightInPoints((short) 12);

		font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);

		// 把字体应用到当前的样式

		style.setFont(font);

		// 生成并设置另一个样式

		HSSFCellStyle style2 = workbook.createCellStyle();

		style2.setFillForegroundColor(HSSFColor.LIGHT_YELLOW.index);

		style2.setFillPattern(HSSFCellStyle.SOLID_FOREGROUND);

		style2.setBorderBottom(HSSFCellStyle.BORDER_THIN);

		style2.setBorderLeft(HSSFCellStyle.BORDER_THIN);

		style2.setBorderRight(HSSFCellStyle.BORDER_THIN);

		style2.setBorderTop(HSSFCellStyle.BORDER_THIN);

		style2.setAlignment(HSSFCellStyle.ALIGN_CENTER);

		style2.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);

		// 生成另一个字体

		HSSFFont font2 = workbook.createFont();

		font2.setBoldweight(HSSFFont.BOLDWEIGHT_NORMAL);

		// 把字体应用到当前的样式

		style2.setFont(font2);

		// 声明一个画图的顶级管理器

		HSSFPatriarch patriarch = sheet.createDrawingPatriarch();

		// 定义注释的大小和位置,详见文档

		HSSFComment comment = patriarch.createComment(new HSSFClientAnchor(0, 0, 0, 0, (short) 4, 2, (short) 6, 5));

		// 设置注释内容

		comment.setString(new HSSFRichTextString("可以在POI中添加注释！"));

		// 设置注释作者，当鼠标移动到单元格上是可以在状态栏中看到该内容.

		comment.setAuthor("suker.zhou");

		// 产生表格标题行

		HSSFRow row = sheet.createRow(0);

		for (int i = 0; i < headers.length; i++) {

			HSSFCell cell = row.createCell(i);

			cell.setCellStyle(style);

			HSSFRichTextString text = new HSSFRichTextString(headers[i]);

			cell.setCellValue(text);

		}

		// 遍历集合数据，产生数据行

		Iterator<T> it = dataset.iterator();

		int index = 0;

		HSSFFont font3 = workbook.createFont();

		font3.setColor(HSSFColor.BLUE.index);
		
		while (it.hasNext()) {

			index++;

			row = sheet.createRow(index);

			T t = (T) it.next();

			// 利用反射，根据javabean属性的先后顺序，动态调用getXxx()方法得到属性值

			Field[] fields = t.getClass().getDeclaredFields();
			
			String[] headList=getHeaderFields(t);

			for (int i = 0; i < headList.length; i++) {

				HSSFCell cell = row.createCell(i);

				cell.setCellStyle(style2);

				String fieldName = headList[i];

				String getMethodName = "get"

				+ fieldName.substring(0, 1).toUpperCase()

				+ fieldName.substring(1);

				try {

					Class tCls = t.getClass();

					Method getMethod = tCls.getMethod(getMethodName,

					new Class[] {});

					Object value = getMethod.invoke(t, new Object[] {});

					// 判断值的类型后进行强制类型转换

					String textValue = null;

					if (value instanceof Boolean) {

						boolean bValue = (Boolean) value;

						textValue = "男";

						if (!bValue) {

							textValue = "女";

						}

					} else if (value instanceof Date) {

						Date date = (Date) value;

						SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

						textValue = sdf.format(date);

					} else if (value instanceof byte[]) {

						// 有图片时，设置行高为60px;

						row.setHeightInPoints(60);

						// 设置图片所在列宽度为80px,注意这里单位的一个换算

						sheet.setColumnWidth(i, (int) (35.7 * 80));

						// sheet.autoSizeColumn(i);

						byte[] bsValue = (byte[]) value;

						HSSFClientAnchor anchor = new HSSFClientAnchor(0, 0,

						1023, 255, (short) 6, index, (short) 6, index);

						anchor.setAnchorType(2);

						patriarch.createPicture(anchor, workbook.addPicture(

						bsValue, HSSFWorkbook.PICTURE_TYPE_JPEG));

					} else {

						// 其它数据类型都当作字符串简单处理
						if(value==null){
							textValue = "";
						}else{
							textValue = value.toString();
						}

					}

					// 如果不是图片数据，就利用正则表达式判断textValue是否全部由数字组成

					if (textValue != null) {

						Pattern p = Pattern.compile("^//d+(//.//d+)?$");

						Matcher matcher = p.matcher(textValue);

						if (matcher.matches()) {

							// 是数字当作double处理

							cell.setCellValue(Double.parseDouble(textValue));

						} else {

							HSSFRichTextString richString = new HSSFRichTextString(textValue);

							richString.applyFont(font3);

							cell.setCellValue(richString);

						}

					}

				} catch (SecurityException e) {

					// TODO Auto-generated catch block

					e.printStackTrace();

				} catch (NoSuchMethodException e) {

					// TODO Auto-generated catch block

					e.printStackTrace();

				} catch (IllegalArgumentException e) {

					// TODO Auto-generated catch block

					e.printStackTrace();

				} catch (IllegalAccessException e) {

					// TODO Auto-generated catch block

					e.printStackTrace();

				} catch (InvocationTargetException e) {

					// TODO Auto-generated catch block

					e.printStackTrace();

				} finally {

					// 清理资源
				}
			}
		}

		return workbook;

	}
	
}
